package 中等.栈;

import java.util.ArrayDeque;
import java.util.Arrays;
import java.util.Deque;

/**
 * 给定一个整数数组 asteroids，表示在同一行的行星。
 * 对于数组中的每一个元素，其绝对值表示行星的大小，正负表示行星的移动方向
 * （正表示向右移动，负表示向左移动）。每一颗行星以相同的速度移动。
 * 找出碰撞后剩下的所有行星。碰撞规则：两个行星相互碰撞，较小的行星会爆炸。
 * 如果两颗行星大小相同，则两颗行星都会爆炸。两颗移动方向相同的行星，永远不会发生碰撞。
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/asteroid-collision
 */
public class 行星碰撞_735 {

    public static void main(String[] args) {

        int[] asteroids = {-2, -2, 1, -1};
        System.out.println(Arrays.toString(asteroidCollision(asteroids)));

    }

    /**
     * 用栈模拟
     * 遇到正数时，直接入栈顶
     * 遇到负数时，
     * 如果栈顶的正数 < 负数的绝对值时，将栈顶元素取出，过程中碰到大小一样的正数，
     * 那么负数不入栈，否则栈为空或者栈顶元素为负数，负数入栈顶
     *
     * @param asteroids
     * @return
     */
    public static int[] asteroidCollision(int[] asteroids) {
        Deque<Integer> stack = new ArrayDeque<>();
        for (int asteroid : asteroids) {
            if (asteroid > 0) {
                stack.addFirst(asteroid);
                continue;
            }

            boolean flag = true;
            while (!stack.isEmpty() && stack.peekFirst() > 0 && stack.peekFirst() <= Math.abs(asteroid)) {
                if (stack.peekFirst() == Math.abs(asteroid)) {  // 碰到相同大小的行星
                    stack.pollFirst();
                    flag = false;
                    break;
                }
                stack.pollFirst();
            }

            if ((stack.isEmpty() || stack.peekFirst() < 0) && flag) {
                stack.addFirst(asteroid);
            }
        }

        int[] ans = new int[stack.size()];
        for (int i = 0; i < ans.length; i++) {
            ans[i] = stack.pollLast();
        }
        return ans;
    }

}
